Node / ReticulumProjects / MeshChatX.git / files / tests / backend / ws_contract_helpers.py

Displaying Raw • Download

T8b949e# SPDX-License-Identifier: 0BSD

T8b949e"""Helpers for WebSocket JSON message contract checks."""

Tff7b72from T7ee787__future__ Tff7b72import Te6edf3annotations

Tff7b72import T7ee787json

Tff7b72import T7ee787re

Tff7b72from T7ee787pathlib Tff7b72import Te6edf3Path

Te6edf3_CLIENT_HANDLER_RE Tff7b72= Te6edf3reTff7b72.Td2a8ffcompileTb4b4b4(

Ta5d6ffrTa5d6ff"Ta5d6ff(?:if|elif)Ta5d6ff\Ta5d6ffs+_typeTa5d6ff\Ta5d6ffs*==Ta5d6ff\Ta5d6ffs*[Tffea00\"Ta5d6ff'Ta5d6ff]([^Tffea00\"Ta5d6ff'Ta5d6ff]+)[Tffea00\"Ta5d6ff'Ta5d6ff]Ta5d6ff"Tb4b4b4,

Tb4b4b4)

Te6edf3_WS_SEND_STR_RE Tff7b72= Te6edf3reTff7b72.Td2a8ffcompileTb4b4b4(

Ta5d6ffrTa5d6ff"Ta5d6ffclientTa5d6ff\Ta5d6ff.send_strTa5d6ff\Ta5d6ffs*Ta5d6ff\Ta5d6ff(Ta5d6ff\Ta5d6ffs*jsonTa5d6ff\Ta5d6ff.dumpsTa5d6ff\Ta5d6ffs*Ta5d6ff\Ta5d6ff(Ta5d6ff\Ta5d6ffs*(Ta5d6ff\Ta5d6ff{Ta5d6ff[Ta5d6ff\Ta5d6ffsTa5d6ff\Ta5d6ffS]*?Ta5d6ff\Ta5d6ff})Ta5d6ff\Ta5d6ffs*,?Ta5d6ff\Ta5d6ffs*Ta5d6ff\Ta5d6ff)Ta5d6ff"Tb4b4b4,

Tb4b4b4)

Te6edf3_RNS_LINK_SEND_RE Tff7b72= Te6edf3reTff7b72.Td2a8ffcompileTb4b4b4(

Ta5d6ffrTa5d6ff"Ta5d6ff_rns_link_sendTa5d6ff\Ta5d6ffs*Ta5d6ff\Ta5d6ff(Ta5d6ff\Ta5d6ffs*clientTa5d6ff\Ta5d6ffs*,Ta5d6ff\Ta5d6ffs*(Ta5d6ff\Ta5d6ff{Ta5d6ff[Ta5d6ff\Ta5d6ffsTa5d6ff\Ta5d6ffS]*?Ta5d6ff\Ta5d6ff})Ta5d6ff\Ta5d6ffs*,?Ta5d6ff\Ta5d6ffs*Ta5d6ff\Ta5d6ff)Ta5d6ff"Tb4b4b4,

Tb4b4b4)

Te6edf3_WS_TYPE_LITERAL_RE Tff7b72= Te6edf3reTff7b72.Td2a8ffcompileTb4b4b4(

Ta5d6ffrTa5d6ff"Ta5d6ff[Tffea00\"Ta5d6ff'Ta5d6ff]type[Tffea00\"Ta5d6ff'Ta5d6ff]Ta5d6ff\Ta5d6ffs*:Ta5d6ff\Ta5d6ffs*[Tffea00\"Ta5d6ff'Ta5d6ff]([^Tffea00\"Ta5d6ff'Ta5d6ff]+)[Tffea00\"Ta5d6ff'Ta5d6ff]Ta5d6ff"Tb4b4b4,

Tb4b4b4)

Te6edf3_BROADCAST_CALL_RE Tff7b72= Te6edf3reTff7b72.Td2a8ffcompileTb4b4b4(

Ta5d6ffrTa5d6ff"Ta5d6ff(?:websocket_broadcast|_broadcast_websocket_message|_broadcast_to_websocket_clients)Ta5d6ff\Ta5d6ffs*Ta5d6ff\Ta5d6ff(Ta5d6ff"Tb4b4b4,

Tb4b4b4)

Tff7b72def Td2a8ffextract_client_inbound_typesTb4b4b4(Te6edf3meshchat_pyTb4b4b4: Te6edf3PathTb4b4b4) Tff7b72-Tff7b72> Tffa657listTb4b4b4[Tffa657strTb4b4b4]Tb4b4b4:

Te6edf3text Tff7b72= Te6edf3meshchat_pyTff7b72.Td2a8ffread_textTb4b4b4(Te6edf3encodingTff7b72=Ta5d6ff"Ta5d6ffutf-8Ta5d6ff"Tb4b4b4)

Te6edf3start Tff7b72= Te6edf3textTff7b72.Td2a8fffindTb4b4b4(Ta5d6ff"Ta5d6ffasync def on_websocket_data_receivedTa5d6ff"Tb4b4b4)

Tff7b72if Te6edf3start Tff7b72< T79c0ff0Tb4b4b4:

Tff7b72return Tb4b4b4[Tb4b4b4]

Te6edf3end Tff7b72= Te6edf3textTff7b72.Td2a8fffindTb4b4b4(Ta5d6ff"Tffea00\nTa5d6ff async def Ta5d6ff"Tb4b4b4, Te6edf3start Tff7b72+ T79c0ff1Tb4b4b4)

Tff7b72if Te6edf3end Tff7b72< T79c0ff0Tb4b4b4:

Te6edf3end Tff7b72= Te6edf3textTff7b72.Td2a8fffindTb4b4b4(Ta5d6ff"Tffea00\nTa5d6ff def Ta5d6ff"Tb4b4b4, Te6edf3start Tff7b72+ T79c0ff1Tb4b4b4)

Te6edf3block Tff7b72= Te6edf3textTb4b4b4[Te6edf3startTb4b4b4:Te6edf3endTb4b4b4] Tff7b72if Te6edf3end Tff7b72> Te6edf3start Tff7b72else Te6edf3textTb4b4b4[Te6edf3startTb4b4b4:Tb4b4b4]

Te6edf3types Tff7b72= Te6edf3_CLIENT_HANDLER_RETff7b72.Td2a8fffindallTb4b4b4(Te6edf3blockTb4b4b4)

Tff7b72return Tffa657sortedTb4b4b4(Tffa657setTb4b4b4(Te6edf3typesTb4b4b4)Tb4b4b4)

Tff7b72def Td2a8ff_extract_type_literals_from_dict_literalTb4b4b4(Te6edf3blobTb4b4b4: Tffa657strTb4b4b4) Tff7b72-Tff7b72> Tffa657str Tff7b72| Tff7b72NoneTb4b4b4:

Te6edf3match Tff7b72= Te6edf3reTff7b72.Td2a8ffsearchTb4b4b4(Ta5d6ffrTa5d6ff"Ta5d6ff[Tffea00\"Ta5d6ff'Ta5d6ff]type[Tffea00\"Ta5d6ff'Ta5d6ff]Ta5d6ff\Ta5d6ffs*:Ta5d6ff\Ta5d6ffs*[Tffea00\"Ta5d6ff'Ta5d6ff]([^Tffea00\"Ta5d6ff'Ta5d6ff]+)[Tffea00\"Ta5d6ff'Ta5d6ff]Ta5d6ff"Tb4b4b4, Te6edf3blobTb4b4b4)

Tff7b72return Te6edf3matchTff7b72.Td2a8ffgroupTb4b4b4(T79c0ff1Tb4b4b4) Tff7b72if Te6edf3match Tff7b72else Tff7b72None

Tff7b72def Td2a8ffextract_client_direct_response_typesTb4b4b4(Te6edf3meshchat_pyTb4b4b4: Te6edf3PathTb4b4b4) Tff7b72-Tff7b72> Tffa657listTb4b4b4[Tffa657strTb4b4b4]Tb4b4b4:

Te6edf3text Tff7b72= Te6edf3meshchat_pyTff7b72.Td2a8ffread_textTb4b4b4(Te6edf3encodingTff7b72=Ta5d6ff"Ta5d6ffutf-8Ta5d6ff"Tb4b4b4)

T8b949e# Include rns.link.* helpers that live immediately after the WS dispatcher.

Te6edf3start Tff7b72= Te6edf3textTff7b72.Td2a8fffindTb4b4b4(Ta5d6ff"Ta5d6ffasync def on_websocket_data_receivedTa5d6ff"Tb4b4b4)

Tff7b72if Te6edf3start Tff7b72< T79c0ff0Tb4b4b4:

Tff7b72return Tb4b4b4[Tb4b4b4]

Te6edf3end Tff7b72= Te6edf3textTff7b72.Td2a8fffindTb4b4b4(Ta5d6ff"Tffea00\nTa5d6ff async def websocket_broadcastTa5d6ff"Tb4b4b4, Te6edf3start Tff7b72+ T79c0ff1Tb4b4b4)

Tff7b72if Te6edf3end Tff7b72< T79c0ff0Tb4b4b4:

Te6edf3end Tff7b72= Te6edf3textTff7b72.Td2a8fffindTb4b4b4(Ta5d6ff"Tffea00\nTa5d6ff async def Ta5d6ff"Tb4b4b4, Te6edf3start Tff7b72+ T79c0ff1Tb4b4b4)

Te6edf3block Tff7b72= Te6edf3textTb4b4b4[Te6edf3startTb4b4b4:Te6edf3endTb4b4b4] Tff7b72if Te6edf3end Tff7b72> Te6edf3start Tff7b72else Te6edf3textTb4b4b4[Te6edf3startTb4b4b4:Tb4b4b4]

Te6edf3typesTb4b4b4: Tffa657setTb4b4b4[Tffa657strTb4b4b4] Tff7b72= Tffa657setTb4b4b4(Tb4b4b4)

Tff7b72for Te6edf3blob Tff7b72in Te6edf3_WS_SEND_STR_RETff7b72.Td2a8fffindallTb4b4b4(Te6edf3blockTb4b4b4)Tb4b4b4:

Te6edf3msg_type Tff7b72= Te6edf3_extract_type_literals_from_dict_literalTb4b4b4(Te6edf3blobTb4b4b4)

Tff7b72if Te6edf3msg_typeTb4b4b4:

Te6edf3typesTff7b72.Td2a8ffaddTb4b4b4(Te6edf3msg_typeTb4b4b4)

Tff7b72for Te6edf3blob Tff7b72in Te6edf3_RNS_LINK_SEND_RETff7b72.Td2a8fffindallTb4b4b4(Te6edf3blockTb4b4b4)Tb4b4b4:

Te6edf3msg_type Tff7b72= Te6edf3_extract_type_literals_from_dict_literalTb4b4b4(Te6edf3blobTb4b4b4)

Tff7b72if Te6edf3msg_typeTb4b4b4:

Te6edf3typesTff7b72.Td2a8ffaddTb4b4b4(Te6edf3msg_typeTb4b4b4)

Tff7b72return Tffa657sortedTb4b4b4(Te6edf3typesTb4b4b4)

Tff7b72def Td2a8ffextract_server_broadcast_typesTb4b4b4(Te6edf3meshchat_pyTb4b4b4: Te6edf3PathTb4b4b4) Tff7b72-Tff7b72> Tffa657listTb4b4b4[Tffa657strTb4b4b4]Tb4b4b4:

Te6edf3text Tff7b72= Te6edf3meshchat_pyTff7b72.Td2a8ffread_textTb4b4b4(Te6edf3encodingTff7b72=Ta5d6ff"Ta5d6ffutf-8Ta5d6ff"Tb4b4b4)

Te6edf3typesTb4b4b4: Tffa657setTb4b4b4[Tffa657strTb4b4b4] Tff7b72= Tffa657setTb4b4b4(Tb4b4b4)

Tff7b72for Te6edf3match Tff7b72in Te6edf3_BROADCAST_CALL_RETff7b72.Td2a8fffinditerTb4b4b4(Te6edf3textTb4b4b4)Tb4b4b4:

Te6edf3chunk Tff7b72= Te6edf3textTb4b4b4[Te6edf3matchTff7b72.Td2a8ffendTb4b4b4(Tb4b4b4) Tb4b4b4: Te6edf3matchTff7b72.Td2a8ffendTb4b4b4(Tb4b4b4) Tff7b72+ T79c0ff1200Tb4b4b4]

Te6edf3type_match Tff7b72= Te6edf3_WS_TYPE_LITERAL_RETff7b72.Td2a8ffsearchTb4b4b4(Te6edf3chunkTb4b4b4)

Tff7b72if Te6edf3type_matchTb4b4b4:

Te6edf3typesTff7b72.Td2a8ffaddTb4b4b4(Te6edf3type_matchTff7b72.Td2a8ffgroupTb4b4b4(T79c0ff1Tb4b4b4)Tb4b4b4)

Te6edf3link_manager Tff7b72= Te6edf3meshchat_pyTff7b72.Td2a8ffparent Tff7b72/ Ta5d6ff"Ta5d6ffsrcTa5d6ff" Tff7b72/ Ta5d6ff"Ta5d6ffbackendTa5d6ff" Tff7b72/ Ta5d6ff"Ta5d6ffrns_link_manager.pyTa5d6ff"

Tff7b72if Te6edf3link_managerTff7b72.Td2a8ffis_fileTb4b4b4(Tb4b4b4)Tb4b4b4:

Te6edf3link_text Tff7b72= Te6edf3link_managerTff7b72.Td2a8ffread_textTb4b4b4(Te6edf3encodingTff7b72=Ta5d6ff"Ta5d6ffutf-8Ta5d6ff"Tb4b4b4)

Tff7b72for Te6edf3type_match Tff7b72in Te6edf3_WS_TYPE_LITERAL_RETff7b72.Td2a8fffinditerTb4b4b4(Te6edf3link_textTb4b4b4)Tb4b4b4:

Te6edf3msg_type Tff7b72= Te6edf3type_matchTff7b72.Td2a8ffgroupTb4b4b4(T79c0ff1Tb4b4b4)

Tff7b72if Te6edf3msg_typeTff7b72.Td2a8ffstartswithTb4b4b4(Ta5d6ff"Ta5d6ffrns.link.Ta5d6ff"Tb4b4b4)Tb4b4b4:

Te6edf3typesTff7b72.Td2a8ffaddTb4b4b4(Te6edf3msg_typeTb4b4b4)

Tff7b72for Te6edf3fn_name Tff7b72in Tb4b4b4(Ta5d6ff"Ta5d6ffsend_config_to_websocket_clientsTa5d6ff"Tb4b4b4,Tb4b4b4)Tb4b4b4:

Te6edf3start Tff7b72= Te6edf3textTff7b72.Td2a8fffindTb4b4b4(Ta5d6fffTa5d6ff"Ta5d6ffasync def Tffd700{Te6edf3fn_nameTffd700}Ta5d6ff"Tb4b4b4)

Tff7b72if Te6edf3start Tff7b72< T79c0ff0Tb4b4b4:

Tff7b72continue

Te6edf3end Tff7b72= Te6edf3textTff7b72.Td2a8fffindTb4b4b4(Ta5d6ff"Tffea00\nTa5d6ff async def Ta5d6ff"Tb4b4b4, Te6edf3start Tff7b72+ T79c0ff1Tb4b4b4)

Te6edf3block Tff7b72= Te6edf3textTb4b4b4[Te6edf3startTb4b4b4:Te6edf3endTb4b4b4] Tff7b72if Te6edf3end Tff7b72> Te6edf3start Tff7b72else Te6edf3textTb4b4b4[Te6edf3start Tb4b4b4: Te6edf3start Tff7b72+ T79c0ff6000Tb4b4b4]

Tff7b72for Te6edf3type_match Tff7b72in Te6edf3_WS_TYPE_LITERAL_RETff7b72.Td2a8fffinditerTb4b4b4(Te6edf3blockTb4b4b4)Tb4b4b4:

Te6edf3typesTff7b72.Td2a8ffaddTb4b4b4(Te6edf3type_matchTff7b72.Td2a8ffgroupTb4b4b4(T79c0ff1Tb4b4b4)Tb4b4b4)

Tff7b72return Tffa657sortedTb4b4b4(Te6edf3typesTb4b4b4)

Tff7b72def Td2a8ffload_ws_manifestTb4b4b4(Te6edf3fixture_pathTb4b4b4: Te6edf3PathTb4b4b4) Tff7b72-Tff7b72> Tffa657dictTb4b4b4[Tffa657strTb4b4b4, Tffa657listTb4b4b4[Tffa657strTb4b4b4]Tb4b4b4]Tb4b4b4:

Te6edf3data Tff7b72= Te6edf3jsonTff7b72.Td2a8ffloadsTb4b4b4(Te6edf3fixture_pathTff7b72.Td2a8ffread_textTb4b4b4(Te6edf3encodingTff7b72=Ta5d6ff"Ta5d6ffutf-8Ta5d6ff"Tb4b4b4)Tb4b4b4)

Tff7b72for Te6edf3key Tff7b72in Tb4b4b4(Ta5d6ff"Ta5d6ffclient_inboundTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ffclient_direct_responsesTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ffserver_broadcastTa5d6ff"Tb4b4b4)Tb4b4b4:

Te6edf3dataTb4b4b4[Te6edf3keyTb4b4b4] Tff7b72= Tffa657sortedTb4b4b4(Te6edf3dataTff7b72.Td2a8ffgetTb4b4b4(Te6edf3keyTb4b4b4, Tb4b4b4[Tb4b4b4]Tb4b4b4)Tb4b4b4)

Tff7b72return Te6edf3data

Tff7b72def Td2a8ffwrite_ws_manifestTb4b4b4(Te6edf3fixture_pathTb4b4b4: Te6edf3PathTb4b4b4, Te6edf3manifestTb4b4b4: Tffa657dictTb4b4b4[Tffa657strTb4b4b4, Tffa657listTb4b4b4[Tffa657strTb4b4b4]Tb4b4b4]Tb4b4b4) Tff7b72-Tff7b72> Tff7b72NoneTb4b4b4:

Te6edf3fixture_pathTff7b72.Td2a8ffparentTff7b72.Td2a8ffmkdirTb4b4b4(Te6edf3parentsTff7b72=Tff7b72TrueTb4b4b4, Te6edf3exist_okTff7b72=Tff7b72TrueTb4b4b4)

Te6edf3ordered Tff7b72= Tb4b4b4{

Ta5d6ff"Ta5d6ffclient_inboundTa5d6ff"Tb4b4b4: Tffa657sortedTb4b4b4(Te6edf3manifestTff7b72.Td2a8ffgetTb4b4b4(Ta5d6ff"Ta5d6ffclient_inboundTa5d6ff"Tb4b4b4, Tb4b4b4[Tb4b4b4]Tb4b4b4)Tb4b4b4)Tb4b4b4,

Ta5d6ff"Ta5d6ffclient_direct_responsesTa5d6ff"Tb4b4b4: Tffa657sortedTb4b4b4(Te6edf3manifestTff7b72.Td2a8ffgetTb4b4b4(Ta5d6ff"Ta5d6ffclient_direct_responsesTa5d6ff"Tb4b4b4, Tb4b4b4[Tb4b4b4]Tb4b4b4)Tb4b4b4)Tb4b4b4,

Ta5d6ff"Ta5d6ffserver_broadcastTa5d6ff"Tb4b4b4: Tffa657sortedTb4b4b4(Te6edf3manifestTff7b72.Td2a8ffgetTb4b4b4(Ta5d6ff"Ta5d6ffserver_broadcastTa5d6ff"Tb4b4b4, Tb4b4b4[Tb4b4b4]Tb4b4b4)Tb4b4b4)Tb4b4b4,

Tb4b4b4}

Te6edf3fixture_pathTff7b72.Td2a8ffwrite_textTb4b4b4(

Te6edf3jsonTff7b72.Td2a8ffdumpsTb4b4b4(Te6edf3orderedTb4b4b4, Te6edf3indentTff7b72=T79c0ff2Tb4b4b4) Tff7b72+ Ta5d6ff"Tffea00\nTa5d6ff"Tb4b4b4,

Te6edf3encodingTff7b72=Ta5d6ff"Ta5d6ffutf-8Ta5d6ff"Tb4b4b4,

Tb4b4b4)